Prompt Manager Complete + Week 3 Review

def search_prompts(prompts):
    """Filter the library by a keyword and display matching prompts."""
    query = input("\nSearch (keyword or platform name): ").lower()
    # .lower() means searching "kling" and "Kling" both match

    # List comprehension — reads as:
    # "Give me each p from prompts WHERE the query appears in any field"
    # This replaces a for loop + if + append pattern in one readable line
    results = [
        p for p in prompts
        if query in p["platform"].lower()
        or query in p["shot"].lower()
        or query in p["prompt"].lower()
    ]

    if not results:
        print(f"\n  No prompts matching '{query}'.\n")
        return

    print(f"\n--- Found {len(results)} match(es) ---\n")
    # Display results the same way show_all() does
    for i, p in enumerate(results):
        print(f"  {i + 1}. [{p['platform']}] {p['shot']}")
        print(f"     {p['prompt'][:60]}...\n")

def delete_prompt(prompts):
    """Show the full list, ask which number to delete, remove it."""
    if not prompts:
        print("\n  Nothing to delete.\n")
        return

    # Show the numbered list so the user knows what to type
    show_all(prompts)
    choice = input("Delete which number? (or 'cancel'): ")

    if choice.lower() == "cancel":
        return

    # try/except here because int("abc") raises ValueError
    # We never trust raw user input when we need a number
    try:
        # Subtract 1 because the menu shows 1-based numbers,
        # but Python lists are 0-based
        index = int(choice) - 1

        # Guard against negative numbers and numbers beyond list length
        if index < 0 or index >= len(prompts):
            print("  Invalid number.\n")
            return

        # .pop(index) removes the item at that position AND returns it
        # so we can tell the user what was deleted
        removed = prompts.pop(index)
        save_prompts(prompts)
        print(f"\n  Deleted: [{removed['platform']}] {removed['shot']}\n")

    except ValueError:
        # int("abc") → ValueError, caught here instead of crashing
        print("  Please enter a number.\n")

def export_by_platform(prompts):
    """Group prompts by platform and write one JSON file per platform."""
    if not prompts:
        print("\n  Nothing to export.\n")
        return

    # Build a dict like: {"Kling": [...], "Runway": [...], "Veo": [...]}
    # We start with an empty dict and add keys as we encounter new platforms
    grouped = {}
    for p in prompts:
        platform = p["platform"]
        if platform not in grouped:
            # First time we see this platform — create an empty list for it
            grouped[platform] = []
        grouped[platform].append(p)

    # Now write one file per platform
    # .items() gives us (key, value) pairs from the dictionary
    for platform, platform_prompts in grouped.items():
        # Build a safe filename: "prompts_kling.json", "prompts_runway.json", etc.
        filename = f"prompts_{platform.lower()}.json"
        with open(filename, "w") as f:
            json.dump(platform_prompts, f, indent=2)
        print(f"  Exported {len(platform_prompts)} prompts to {filename}")

    print()

prompts = load_prompts()

while True:
    print("\n========== Prompt Manager ==========")
    print("  1. View all prompts")
    print("  2. Add a prompt")
    print("  3. Search prompts")
    print("  4. Delete a prompt")
    print("  5. Export by platform")
    print("  6. Quit")
    print("===================================\n")

    choice = input("Choose (1-6): ").strip()

    if   choice == "1": show_all(prompts)
    elif choice == "2": add_prompt(prompts)
    elif choice == "3": search_prompts(prompts)   # was placeholder
    elif choice == "4": delete_prompt(prompts)    # was placeholder
    elif choice == "5": export_by_platform(prompts) # was placeholder
    elif choice == "6":
        print("\n  Bye!\n")
        break
    else:
        print("  Invalid choice. Enter 1-6.\n")